pandas多行合并成一行

立秋6789

已于 2022-07-09 14:50:05 修改

阅读量4.9k

点赞数 4

分类专栏： pandas 文章标签： pandas python 数据分析

于 2022-07-09 14:33:18 首次发布

本文链接：https://blog.csdn.net/summerriver1/article/details/125693263

版权

pandas 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

文章目录

构建df

import pandas as pd

df = pd.DataFrame(
    {'水果': ['苹果', '香蕉', '西瓜', '葡萄', '柚子', '芒果'], '公司': ['A1', 'A1', 'A2', 'A2', 'A2', 'A3'], '重量': [1, 2, 3, 4, 5, 6],
     '数量': [3, 2, 5, 10, 10, 6]})

df结构

请添加图片描述

把相同公司的行合并成一行

print(df.groupby(['公司']).agg({'水果': [', '.join], '数量': lambda x: list(x),
                        '重量': lambda x: list(x)}).reset_index())

效果

请添加图片描述

代码汇总

# pandas多列合并成一行
import pandas as pd

df = pd.DataFrame(
    {'水果': ['苹果', '香蕉', '西瓜', '葡萄', '柚子', '芒果'], '公司': ['A1', 'A1', 'A2', 'A2', 'A2', 'A3'], '重量': [1, 2, 3, 4, 5, 6],
     '数量': [3, 2, 5, 10, 10, 6]})
# print(df)
print(df.groupby(['公司']).agg({'水果': [', '.join], '数量': lambda x: list(x),
                        '重量': lambda x: list(x)}).reset_index())